tf-nightly-gpu
transformers
absl-py
colorama
numpy
tqdm
datasets
psutil
h5py
